class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int n = nums.size();
        vector<int> f(n, 0);    // 前缀和数组
        vector<int> g(n, 0);    // 后缀和数组

        f[0] = 0, g[n - 1] = 0;
        for(int i = 1; i < n; i++)
            f[i] = f[i - 1] + nums[i - 1];  // 前缀和

        for(int i = n - 2; i >= 0; i--)
            g[i] = g[i + 1] + nums[i + 1];  // 后缀和

        for(int i = 0; i < n; i++)
        {
            if(f[i] == g[i])
                return i;
        }
        return -1;
    }
};